An AI4AI Framework for Visual Token Pruning
视觉 Token 剪枝的 AI4AI 框架
🧭 LLM 驱动的视觉 Token 剪枝策略自动设计
Xi'an Jiaotong University · Ningbo University of Technology · North China University of Technology · MiLM Plus, Xiaomi Inc. · Hong Kong University of Science and Technology
提出 AutoPrune,用 LLM 驱动 TPDSL 自动设计视觉 token 剪枝策略,免训练,删 94.4% token 仍保持 99% 以上性能
📌 工作参考价值(4 点)
- 多模态推理降本:电商商品图/视频理解、广告创意理解等多模态服务中,可借鉴 AutoPrune 的训练-free 剪枝思路,用领域特定语言描述预算控制、token 打分、选择约束,快速替代手工规则,降低 prefill 延迟。
- 残差式策略搜索:不从头搜索,而是在强基础策略上做残差修改,缩小搜索空间并让 LLM 聚焦关键组件;该思想可迁移到推荐系统策略搜索、特征选择、排序规则自动调优。
- 可复用原子库:将剪枝/召回/过滤等策略拆成可组合原子操作,跨模型和场景迁移;业务中可构建相似 DSL 与原子库,降低策略迭代成本。
- 推理加速指标参考:删 94.4% 视觉 token 仍保持 99% 性能,FLOPs 降 9.9×,prefill 延迟降 6.4×;对低延迟电商搜索/推荐中的多模态 LLM 服务有直接工程参考价值。